home *** CD-ROM | disk | FTP | other *** search
/ The Programmer Disk / The Programmer Disk (Microforum).iso / xpro / c4 / pro20 / pbmmask.c < prev    next >
C/C++ Source or Header  |  1990-06-03  |  4KB  |  152 lines
  1. /* pbmmask.c - create a mask bitmap from a portable bitmap
  2. **
  3. ** Copyright (C) 1988 by Jef Poskanzer.
  4. **
  5. ** Permission to use, copy, modify, and distribute this software and its
  6. ** documentation for any purpose and without fee is hereby granted, provided
  7. ** that the above copyright notice appear in all copies and that both that
  8. ** copyright notice and this permission notice appear in supporting
  9. ** documentation.  This software is provided "as is" without express or
  10. ** implied warranty.
  11. */
  12.  
  13. #include <stdio.h>
  14. #include "pbm.h"
  15.  
  16. bit **bits, **mask, backcolor;
  17. int rows, cols;
  18.  
  19. main( argc, argv )
  20. int argc;
  21. char *argv[];
  22.     {
  23.     FILE *ifd;
  24.     register int row, col, wcount;
  25.  
  26.     pm_progname = argv[0];
  27.  
  28.     if ( argc > 2 )
  29.     pm_usage( "[pbmfile]" );
  30.  
  31.     if ( argc == 2 )
  32.     ifd = pm_openr( argv[1] );
  33.     else
  34.     ifd = stdin;
  35.  
  36.     bits = pbm_readpbm( ifd, &cols, &rows );
  37.     pm_close( ifd );
  38.     mask = pbm_allocarray( cols, rows );
  39.  
  40.     /* Clear out the mask. */
  41.     for ( row = 0; row < rows; row++ )
  42.         for ( col = 0; col < cols; col++ )
  43.         mask[row][col] = PBM_BLACK;
  44.  
  45.     /* Figure out the background color, by counting along the edge. */
  46.     wcount = 0;
  47.     for ( row = 0; row < rows; row++ )
  48.     {
  49.     if ( bits[row][0] == PBM_WHITE )
  50.         wcount++;
  51.     if ( bits[row][cols - 1] == PBM_WHITE )
  52.         wcount++;
  53.     }
  54.     for ( col = 1; col < cols - 1; col++ )
  55.     {
  56.     if ( bits[0][col] == PBM_WHITE )
  57.         wcount++;
  58.     if ( bits[rows - 1][col] == PBM_WHITE )
  59.         wcount++;
  60.     }
  61.     if ( wcount >= rows + cols - 2 )
  62.     backcolor = PBM_WHITE;
  63.     else
  64.     backcolor = PBM_BLACK;
  65.  
  66.     /* Flood the entire edge.  Probably the first call will be enough, but
  67.     ** might as well be sure. */
  68.     for ( col = cols - 3; col >= 2; col -= 2 )
  69.     {
  70.     addflood( col, rows - 1 );
  71.     addflood( col, 0 );
  72.     }
  73.     for ( row = rows - 1; row >= 0; row -= 2 )
  74.     {
  75.     addflood( cols - 1, row );
  76.     addflood( 0, row );
  77.     }
  78.     flood( );
  79.  
  80.     /* All done. */
  81.     pbm_writepbm( stdout, mask, cols, rows );
  82.  
  83.     exit( 0 );
  84.     }
  85.  
  86. #ifdef MSDOS
  87. #define FLOODSTACKSIZE 5000
  88. #else
  89. #define FLOODSTACKSIZE 50000
  90. #endif
  91. short fcols[FLOODSTACKSIZE], frows[FLOODSTACKSIZE];
  92. int fstackp = 0;
  93.  
  94. addflood( col, row )
  95. int col, row;
  96.     {
  97.     if ( bits[row][col] == backcolor && mask[row][col] == PBM_BLACK )
  98.     {
  99.     if ( fstackp >= FLOODSTACKSIZE )
  100.         pm_error( "flood stack overflow", 0,0,0,0,0 );
  101.     fcols[fstackp] = col;
  102.     frows[fstackp] = row;
  103.     fstackp++;
  104.     }
  105.     }
  106.  
  107. flood( )
  108.     {
  109.     register int col, row, c;
  110.  
  111.     while ( fstackp > 0 )
  112.     {
  113.     fstackp--;
  114.     col = fcols[fstackp];
  115.     row = frows[fstackp];
  116.     if ( bits[row][col] == backcolor && mask[row][col] == PBM_BLACK )
  117.         {
  118.         mask[row][col] = PBM_WHITE;
  119.         if ( row - 1 >= 0 )
  120.         addflood( col, row - 1 );
  121.         if ( row + 1 < rows )
  122.         addflood( col, row + 1 );
  123.         for ( c = col + 1; c < cols; c++ )
  124.         {
  125.         if ( bits[row][c] == backcolor && mask[row][c] == PBM_BLACK )
  126.             {
  127.             mask[row][c] = PBM_WHITE;
  128.             if ( row - 1 >= 0 && ( bits[row - 1][c - 1] != backcolor || mask[row - 1][c - 1] != PBM_BLACK ) )
  129.             addflood( c, row - 1 );
  130.             if ( row + 1 < rows && ( bits[row + 1][c - 1] != backcolor || mask[row + 1][c - 1] != PBM_BLACK ) )
  131.             addflood( c, row + 1 );
  132.             }
  133.         else
  134.             break;
  135.         }
  136.         for ( c = col - 1; c >= 0; c-- )
  137.         {
  138.         if ( bits[row][c] == backcolor && mask[row][c] == PBM_BLACK )
  139.             {
  140.             mask[row][c] = PBM_WHITE;
  141.             if ( row - 1 >= 0 && ( bits[row - 1][c + 1] != backcolor || mask[row - 1][c + 1] != PBM_BLACK ) )
  142.             addflood( c, row - 1 );
  143.             if ( row + 1 < rows && ( bits[row + 1][c + 1] != backcolor || mask[row + 1][c + 1] != PBM_BLACK ) )
  144.             addflood( c, row + 1 );
  145.             }
  146.         else
  147.             break;
  148.         }
  149.         }
  150.     }
  151.     }
  152.